package com.xh.hongaiagent.rag;

import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * 上下文查询增强器
 */
public class LoveAppContextualQueryAugmenterFactory {
    public static ContextualQueryAugmenter createLoveAppContextualQueryAugmenter(){
        PromptTemplate emptyContextPromptTemplate = new PromptTemplate("""
                你应该回答：
                抱歉，我只能回答恋爱相关的问题，别的问题我也不知道。
                """);
        return new ContextualQueryAugmenter.Builder()
                .allowEmptyContext(true)
                .emptyContextPromptTemplate(emptyContextPromptTemplate)
                .build();
    }
}
